import torch
import os
# APP服务
host = "0.0.0.0"
port = 32224

# device_type = "cpu"  # cuda:0、cuda:1、cpu
device_type = "cuda" if torch.cuda.is_available() else "cpu"

# input_Excel_path = "input/热线数据.xlsx"
input_Excel_path = "input/test.xlsx"

output_Excel_path = "output/test/0117_lhs_test1.xlsx"

model_version = "/yfzx/models/embeddings/m3e-large"

Is_single = True

#设置lsh分配的平面数量[影响运行速度，数量越多，速度越快，准确率会下降]
lsh_num_planes = 10

control_output_num = 1

threshold = 0.7

reunion_threshold = 0.7

find_similar_threshold = 0.95

log_path = './logs/service_test_similarOrder1120.log'

# db_host = '10.242.182.94'
# db_port = 63302
# db_user = 'root'
# db_password = 'b8be81bb627d'
# db_database_admin = 'chengyun'


db_host = "172.16.2.162"
db_user = "bigmodel"
db_password = "Bigmodel20231103"
db_database_admin = "bigmodeldb"
db_port = 3308

#sql读出来的表格式固定为 工单id | 地址 | 工单内容 | 分类类型
# read_sql_data = """select taskid, address, description, infoatname from ch_work_order_slim where discovertime >= DATE_SUB(CURDATE(), INTERVAL 3 DAY);"""  # 需要修改，手动添加

# read_sql_data = f"select * from T_GD2 where position(工单编号 in {data})"


# LLM_MODEL_DICT = {
#     "chatglm3-6b":"/data/model/chatglm3-6b"
# }


PROJECT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
# output = f"{num}人反映{event}问题"